package com.kamistoat.ruoyi.ruoyicommoncore.utils.reflect;

import com.kamistoat.ruoyi.ruoyicommoncore.text.Convert;
import com.kamistoat.ruoyi.ruoyicommoncore.utils.DateUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;
import org.apache.poi.ss.usermodel.DateUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.*;
import java.util.Date;

/**
 * 反射工具类.
 * 提供在运行期间: 父子类间多级调用getter/setter, 访问私有变量, 调用私有方法, 获取泛型类型Class, 被AOP过的真实类等工具函数.
 * 非常重要，理解代码!!!
 */
@SuppressWarnings("rawtypes")
public class ReflectUtils {
    private static final String SETTER_PREFIX = "set";

    private static final String GETTER_PREFIX = "get";

    private static final String CGLIB_CLASS_SEPARATOR = "$$";

    private static final Logger logger = LoggerFactory.getLogger(ReflectUtils.class);

    /**
     * 调用Getter方法.
     *
     * @param obj          当前实体类。
     * @param propertyName 要在实体类的哪个属性字段上执行get方法?
     *                     支持多级，如： 属性字段.(中的)属性字段
     */
    @SuppressWarnings("unchecked")
    public static <E> E invokeGetter(Object obj, String propertyName) {
        Object object = obj;
        // 将 propertyName按照 . 分割开，然后遍历分割得到的每一个内容
        for (String name : StringUtils.split(propertyName, ".")) {
            // 允许多级调用，从 ExcelUtil传来的有可能是 dept.deptName 的形式
            // 则for首先构建 getDept()，经过下面的 invokeMethod()方法得到object(即dept对象)，然后再去这个object中 getDeptName()，得到最终的object(即String类)
            String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(name);
            // 三参四参非null，只是空数组。因为get方法无输入参数
            object = invokeMethod(object, getterMethodName, new Class[]{}, new Object[]{});
        }
        return (E) object;
    }

    /**
     * 从obj源类开始向其父类寻找目标方法methodName, 返回第一个匹配到的方法调用结果，且强行无视private/protected修饰符.
     * 同时匹配方法名+参数类型。
     *
     * @param obj            一个实体类，并非是元类
     * @param methodName     想要调用实体类中的哪个方法?无视方法的修饰符
     * @param parameterTypes Class<?>[] 这个方法的输入参数的元类数组。
     * @param args           和上面数组等长且一一对应的参数数组
     */
    @SuppressWarnings("unchecked")
    public static <E> E invokeMethod(final Object obj,
                                     final String methodName,
                                     final Class<?>[] parameterTypes,
                                     final Object[] args) {

        if (obj == null || methodName == null) {
            return null;
        }

        // 从当前的 obj实体类开始，寻找其本身或父类中第一个符合methodName+parameterTypes的方法
        Method method = getAccessibleMethod(obj, methodName, parameterTypes);

        // 全部找完也没有找到
        if (method == null) {
            logger.debug("在 [" + obj.getClass() + "] 及其父类中，没有找到 [" + methodName + "] 方法 ");
            return null;
        }

        // 如果找到了这个方法，则调用反射库包中的 method.invoke()，传入当前实体类和方法输入参数
        // 传入当前实体类很好理解，因为这个方法可能会用到实体类中其他的属性字段，这些属性字段的值是只和当前实体类有关而和元类无关的
        try {
            return (E) method.invoke(obj, args);
        } catch (Exception e) {
            String msg = "method: " + method + ", obj: " + obj + ", args: " + args + "";
            throw convertReflectionExceptionToUnchecked(msg, e);
        }
    }

    /**
     * 循环向上转型, 获取对象的第一个DeclaredMethod, 并强制设置为可访问.
     * 如向上转型到Object类(即所有对象的父类)仍无法找到, 返回null.
     * 匹配函数名+参数类型。
     * 用于方法需要被多次调用的情况. 先使用本函数先取得Method, 然后调用Method.invoke(Object obj, Object... args)
     *
     * @param obj            当前实体类
     * @param methodName     想要寻找的方法
     * @param parameterTypes 参数
     */
    public static Method getAccessibleMethod(final Object obj, final String methodName,
                                             final Class<?>... parameterTypes) {
        // 为空不报错。直接返回 null
        if (obj == null) {
            return null;
        }

        Validate.notBlank(methodName, "methodName can't be blank");

        /**
         * obj.getClass()--获取当前实体类的元类信息. 反射库只能在元类上，不能在实例上。
         * Object.class--搜索的终止条件为到达元类Object. Object类是所有类的顶层父类
         * Class<?>.getSuperclass()--递进条件为继续获取父类向上找
         */
        for (Class<?> searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) {
            try {
                // 根据 方法名 和 输入参数类型 去找方法。这也是定位一个方法的唯一途径。
                Method method = searchType.getDeclaredMethod(methodName, parameterTypes);
                // 如果能找到这个方法，则将其设置为可访问，然后直接将这个方法返回，不再继续向上寻找
                makeAccessible(method);
                return method;
            } catch (NoSuchMethodException e) {
                continue;
            }
        }
        return null;
    }

    /**
     * 改变private/protected的方法为public，尽量不调用实际改动的语句，避免JDK的SecurityManager抱怨。
     */
    public static void makeAccessible(Method method) {
        // method.getModifiers()获取方法的访问权限
        // method.getDeclaringClass().getModifiers()获取方法所在的类的访问权限
        if ((!Modifier.isPublic(method.getModifiers()) || !Modifier.isPublic(method.getDeclaringClass().getModifiers()))
                && !method.isAccessible()) {
            method.setAccessible(true);
        }
    }

    /**
     * 调用Setter方法, 仅匹配方法名。
     * 对于一个entity对象，如果要set多级属性，则先用getter从最顶层属性一直到达最底层属性，然后调用setter重新赋值
     *
     * @param obj          实体类。
     * @param propertyName 要在实体类的哪个属性字段上执行set方法?
     *                     支持多级，如： 属性字段.(中的)属性字段
     * @param value        要set的新值
     */
    public static <E> void invokeSetter(Object obj, String propertyName, E value) {
        Object object = obj;
        // 将多级字段拆开。names[0]是最顶级字段
        String[] names = StringUtils.split(propertyName, ".");
        for (int i = 0; i < names.length; i++) {
            // 如果没有到达倒数第二底层的字段，则都是先用getter
            // 例如要setter A.B.C，则if会一直getter到 B
            if (i < names.length - 1) {
                String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(names[i]);
                object = invokeMethod(object, getterMethodName, new Class[]{}, new Object[]{});
            } else {
                // 然后从倒数第二层字段开始再调用setter来重新赋值
                String setterMethodName = SETTER_PREFIX + StringUtils.capitalize(names[i]);
                invokeMethodByName(object, setterMethodName, new Object[]{value});
            }
        }
    }

    /**
     * 直接调用对象方法, 无视private/protected修饰符，
     * 用于一次性调用的情况，否则应使用getAccessibleMethodByName()函数获得Method后反复调用.
     * 只匹配函数名，如果有多个同名函数调用第一个。
     * <p>
     * 只在获取setter方法中使用过。增加了数据类型转换的代码
     */
    @SuppressWarnings("unchecked")
    public static <E> E invokeMethodByName(final Object obj, final String methodName, final Object[] args) {
        Method method = getAccessibleMethodByName(obj, methodName, args.length);
        if (method == null) {
            // 如果为空不报错，直接返回空。
            logger.debug("在 [" + obj.getClass() + "] 及其父类中，没有找到 [" + methodName + "] 方法 ");
            return null;
        }
        // 找到方法，则开始进行set新值的操作，其实就是把 Object[]中的输入参数转换成 setter方法接收的参数类型
        try {
            // 类型转换（将参数数据类型转换为目标方法参数类型）
            Class<?>[] cs = method.getParameterTypes();
            for (int i = 0; i < cs.length; i++) {
                if (args[i] != null && !args[i].getClass().equals(cs[i])) {
                    if (cs[i] == String.class) {
                        args[i] = Convert.toStr(args[i]);
                        if (StringUtils.endsWith((String) args[i], ".0")) {
                            args[i] = StringUtils.substringBefore((String) args[i], ".0");
                        }
                    } else if (cs[i] == Integer.class) {
                        args[i] = Convert.toInt(args[i]);
                    } else if (cs[i] == Long.class) {
                        args[i] = Convert.toLong(args[i]);
                    } else if (cs[i] == Double.class) {
                        args[i] = Convert.toDouble(args[i]);
                    } else if (cs[i] == Float.class) {
                        args[i] = Convert.toFloat(args[i]);
                    } else if (cs[i] == Date.class) {
                        if (args[i] instanceof String) {
                            args[i] = DateUtils.parseDate(args[i]);
                        } else {
                            args[i] = DateUtil.getJavaDate((Double) args[i]);
                        }
                    } else if (cs[i] == boolean.class || cs[i] == Boolean.class) {
                        args[i] = Convert.toBool(args[i]);
                    }
                }
            }
            return (E) method.invoke(obj, args);
        } catch (Exception e) {
            String msg = "method: " + method + ", obj: " + obj + ", args: " + args + "";
            throw convertReflectionExceptionToUnchecked(msg, e);
        }
    }

    /**
     * 循环向上转型, 获取对象的DeclaredMethod,并强制设置为可访问.
     * 如向上转型到Object仍无法找到, 返回null.
     * 不同与之前的 getAccessibleMethod()方法，该方法只匹配函数名，并不要求函数的输入参数类型也匹配。。
     * 用于方法需要被多次调用的情况. 先使用本函数先取得Method,然后调用Method.invoke(Object obj, Object... args)
     */
    public static Method getAccessibleMethodByName(final Object obj, final String methodName, int argsNum) {
        // 为空不报错。直接返回 null
        if (obj == null) {
            return null;
        }
        Validate.notBlank(methodName, "methodName can't be blank");
        for (Class<?> searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) {
            Method[] methods = searchType.getDeclaredMethods();
            for (Method method : methods) {
                if (method.getName().equals(methodName) && method.getParameterTypes().length == argsNum) {
                    makeAccessible(method);
                    return method;
                }
            }
        }
        return null;
    }

    /**
     * 直接读取对象属性值, 无视private/protected修饰符, 不经过getter函数.
     * 注，对于private属性原本只能通过写 public的getter方法来获取，这里是强制获取
     *
     * @param obj       当前类实例
     * @param fieldName 想要获取的属性字段的名字
     */
    @SuppressWarnings("unchecked")
    public static <E> E getFieldValue(final Object obj, final String fieldName) {
        Field field = getAccessibleField(obj, fieldName);
        if (field == null) {
            logger.debug("在 [" + obj.getClass() + "] 中，没有找到 [" + fieldName + "] 字段 ");
            return null;
        }
        E result = null;
        try {
            result = (E) field.get(obj);
        } catch (IllegalAccessException e) {
            logger.error("不可能抛出的异常{}", e.getMessage());
        }
        return result;
    }

    /**
     * 直接设置对象属性值, 无视private/protected修饰符, 不经过setter函数.
     */
    public static <E> void setFieldValue(final Object obj, final String fieldName, final E value) {
        Field field = getAccessibleField(obj, fieldName);
        if (field == null) {
            // throw new IllegalArgumentException("在 [" + obj.getClass() + "] 中，没有找到 [" + fieldName + "] 字段 ");
            logger.debug("在 [" + obj.getClass() + "] 中，没有找到 [" + fieldName + "] 字段 ");
            return;
        }
        try {
            field.set(obj, value);
        } catch (IllegalAccessException e) {
            logger.error("不可能抛出的异常: {}", e.getMessage());
        }
    }

    /**
     * 循环向上转型, 获取对象的DeclaredField, 并强制设置为可访问.
     * 如向上转型到Object仍无法找到, 返回null.
     */
    public static Field getAccessibleField(final Object obj, final String fieldName) {
        // 为空不报错。直接返回 null
        if (obj == null) {
            return null;
        }
        Validate.notBlank(fieldName, "fieldName can't be blank");
        for (Class<?> superClass = obj.getClass(); superClass != Object.class; superClass = superClass.getSuperclass()) {
            try {
                Field field = superClass.getDeclaredField(fieldName);
                makeAccessible(field);
                return field;
            } catch (NoSuchFieldException e) {
                continue;
            }
        }
        return null;
    }

    /**
     * 改变private/protected的成员变量为public，尽量不调用实际改动的语句，避免JDK的SecurityManager抱怨。
     */
    public static void makeAccessible(Field field) {
        if ((!Modifier.isPublic(field.getModifiers()) || !Modifier.isPublic(field.getDeclaringClass().getModifiers())
                || Modifier.isFinal(field.getModifiers())) && !field.isAccessible()) {
            field.setAccessible(true);
        }
    }

    /**
     * 通过反射, 获得Class定义中声明的泛型参数的类型, 注意泛型必须定义在父类处
     * 如无法找到, 返回Object.class.
     */
    @SuppressWarnings("unchecked")
    public static <T> Class<T> getClassGenricType(final Class clazz) {
        return getClassGenricType(clazz, 0);
    }

    /**
     * 通过反射, 获得Class定义中声明的父类的泛型参数的类型.
     * 如无法找到, 返回Object.class.
     */
    public static Class getClassGenricType(final Class clazz, final int index) {
        Type genType = clazz.getGenericSuperclass();

        if (!(genType instanceof ParameterizedType)) {
            logger.debug(clazz.getSimpleName() + "'s superclass not ParameterizedType");
            return Object.class;
        }

        Type[] params = ((ParameterizedType) genType).getActualTypeArguments();

        if (index >= params.length || index < 0) {
            logger.debug("Index: " + index + ", Size of " + clazz.getSimpleName() + "'s Parameterized Type: "
                    + params.length);
            return Object.class;
        }
        if (!(params[index] instanceof Class)) {
            logger.debug(clazz.getSimpleName() + " not set the actual class on superclass generic parameter");
            return Object.class;
        }

        return (Class) params[index];
    }

    public static Class<?> getUserClass(Object instance) {
        if (instance == null) {
            throw new RuntimeException("Instance must not be null");
        }
        Class clazz = instance.getClass();
        if (clazz != null && clazz.getName().contains(CGLIB_CLASS_SEPARATOR)) {
            Class<?> superClass = clazz.getSuperclass();
            if (superClass != null && !Object.class.equals(superClass)) {
                return superClass;
            }
        }
        return clazz;

    }

    /**
     * 将反射时的checked exception转换为unchecked exception.
     */
    public static RuntimeException convertReflectionExceptionToUnchecked(String msg, Exception e) {
        if (e instanceof IllegalAccessException || e instanceof IllegalArgumentException
                || e instanceof NoSuchMethodException) {
            return new IllegalArgumentException(msg, e);
        } else if (e instanceof InvocationTargetException) {
            return new RuntimeException(msg, ((InvocationTargetException) e).getTargetException());
        }
        return new RuntimeException(msg, e);
    }


}
